# -*- coding: utf-8 -*-
# @Time    : 2025/2/11 20:49
# @Author  : 
# @File    : download.py
# @Software: PyCharm 
# @Comment : 下载模型和数据集
import os

from huggingface_hub import snapshot_download

from config import Config

# os.environ['HF_ENDPOINT'] = 'https://hf-mirror.com'

snapshot_download(
    repo_id=Config.model_name,
    repo_type="model",
    local_dir=Config.pretrained_model_path
)

snapshot_download(
    repo_id=Config.dataset_name,
    repo_type="dataset",
    local_dir=Config.raw_datasets_path
)
